from datasets import load_dataset
import os
os.environ['HF_DATASETS_OFFLINE'] = '1'

def getflickr30k():
    print("loding dataset nlphuji/flickr30k")
    ds = load_dataset("nlphuji/flickr30k",cache_dir="/root/autodl-tmp/hfcache/datasets/nlphuji/flickr30k")
    print("Done.")
    return ds

def getVQAv2():
    print("loding dataset HuggingFaceM4/VQAv2")
    ds = load_dataset("HuggingFaceM4/VQAv2",cache_dir="/root/autodl-tmp/hfcache/datasets/HuggingFaceM4/VQAv2")
    print("Done.")
    return ds
